import requests, os
path_a = os.path.abspath('.')
kw = ''
while True:
    kw = input('请输入你要获取的图片(若想结束请输入1)')
    if kw == '1':
    	print('已退出，你下载的图片已保存在'+path_a+',请查看！')
    	break
    for x in range(0, 1000, 20):
    url = 'https://www.toutiao.com/search_content/?offset='+str(x)+'&format=json&keyword=%s&autoload=true&count=20&cur_tab=3&from=gallery' % kw
    response = requests.get(url)
    data = response.json()['data']
    if not data:
    	print('下载'+kw+'图片完毕，请换个关键词继续')
    	break
    n = 1  # 记录文章数
    for atlas in data:# 创建目录
    	title = atlas['title']
    	print(atlas)
    	try:
    		if title not in os.listdir('.'):# 防止文件名已经存在
    			os.mkdir(title)
    	except OSError as e:
    		print('文件名出错，创建目录失败，重新创建一个随机名字')
    		title = kw + '文件名出错'+str(x)
    		if title not in os.listdir('.'):
    			os.mkdir(title)
    	k = 1  # 记录下载的图片数
    	path = os.path.join(path_a, title)# 转进图片目录
    	os.chdir(path)
    	for image in atlas['image_list']:  # 这个链接获取的图片是小张的，看着不够爽，所以下面替换成大的图片
    		image_url = image['url'].replace('list', 'large')  # 改个链接获取大的图片
    		atlas = requests.get('http:'+image_url).content
    		with open(str(k)+'.jpg', 'wb') as f:  # 把图片写入文件内
    			f.write(atlas)
    	print('下载完第%d个文章的%d幅图完成' % (x+n, k))
    	k += 1
    n += 1# 转出图片目录            
    os.chdir(path_a)